package com.ssyu.examination.conmmon;

import java.io.File;
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.Date;

/**
 * MySQL数据库备份
 * 
 * @author GaoHuanjie
 */
public class MySQLDatabaseBackup {

	/**
	 * Java代码实现MySQL数据库导出
	 * 
	 * @author GaoHuanjie
	 * @param hostIP
	 *            MySQL数据库所在服务器地址IP
	 * @param userName
	 *            进入数据库所需要的用户名
	 * @param password
	 *            进入数据库所需要的密码
	 * @param savePath
	 *            数据库导出文件保存路径
	 * @param fileName
	 *            数据库导出文件文件名
	 * @param databaseName
	 *            要导出的数据库名
	 * @return 返回true表示导出成功，否则返回false。
	 */
	public static boolean exportDatabaseTool(String hostIP, String userName,
			String password, String savePath, String fileName,
			String databaseName) {
		File saveFile = new File(savePath);
		if (!saveFile.exists()) {// 如果目录不存在
			saveFile.mkdirs();// 创建文件夹
		}
		if (!savePath.endsWith(File.separator)) {
			savePath = savePath + File.separator;
		}

		StringBuilder stringBuilder = new StringBuilder();
		stringBuilder.append("mysqldump").append(" --opt").append(" -h")
				.append(hostIP);
		stringBuilder.append(" --user=").append(userName)
				.append(" --password=").append(password)
				.append(" --lock-all-tables=true");
		stringBuilder.append(" --result-file=").append(savePath + fileName)
				.append(" --default-character-set=utf8 ").append(databaseName);
		try {
			Process process = Runtime.getRuntime().exec(
					stringBuilder.toString());
			if (process.waitFor() == 0) {// 0 表示线程正常终止。
				return true;
			}
		} catch (IOException e) {
			e.printStackTrace();
		} catch (InterruptedException e) {
			e.printStackTrace();
		}
		return false;
	}

	public String getSQLName() {
		Date now = new Date();
		SimpleDateFormat sdf = new SimpleDateFormat("YYYY-MM-dd__hh-mm-ss");
		String str = sdf.format(now) + ".sql";
		System.out.println(str);
		return str;
	}

	private static void FileRemove() {
		File file = new File("D:/backupDatabase");
		String files[];
		files = file.list();
		if (files.length <= 0) {
			return;
		} else if (files.length >= 10) {
			for (int i = files.length; i >= 10; i--) {
				deleteFile("D:/backupDatabase/" + files[i - 10]);
			}
		} else {
			return;
		}
	}

	public static boolean deleteFile(String sPath) {
		boolean flag = false;
		File file = new File(sPath);
		// 路径为文件且不为空则进行删除
		if (file.isFile() && file.exists()) {
			file.delete();
			flag = true;
		}
		return flag;
	}

	public void executeBackup() {

		exportDatabaseTool("127.0.0.1", "root", "root", "D:/backupDatabase",
				getSQLName(), "online_exam");

		FileRemove();
	}

	public static void main(String[] args) throws InterruptedException {
		MySQLDatabaseBackup mySQLDatabaseBackup = new MySQLDatabaseBackup();
		mySQLDatabaseBackup.executeBackup();
	}
}